package com.algorithm.greedy;

import java.util.*;

/**
 * 贪心算法
 */
public class GreedyAlgorithm {
	public static void main(String[] args) {
		//创建广播电台,放入到 Map
		HashMap<String, HashSet<String>> broadcasts = new HashMap();

		//将各个电台放入到 broadcasts
		HashSet<String> hashSet1 = new HashSet<>();
		hashSet1.add("北京");
		hashSet1.add("上海");
		hashSet1.add("天津");
		HashSet<String> hashSet2 = new HashSet<>();
		hashSet2.add("广州");
		hashSet2.add("北京");
		hashSet2.add("深圳");
		HashSet<String> hashSet3 = new HashSet<>();
		hashSet3.add("成都");
		hashSet3.add("上海");
		hashSet3.add("杭州");
		HashSet<String> hashSet4 = new HashSet<>();
		hashSet4.add("上海");
		hashSet4.add("天津");
		HashSet<String> hashSet5 = new HashSet<>();
		hashSet5.add("杭州");
		hashSet5.add("大连");

		//加入到 map
		broadcasts.put("K1", hashSet1);
		broadcasts.put("K2", hashSet2);
		broadcasts.put("K3", hashSet3);
		broadcasts.put("K4", hashSet4);
		broadcasts.put("K5", hashSet5);

		//allAreas 存放所有的地区
		HashSet<String> allAreas = new HashSet<>();
		allAreas.add("北京");
		allAreas.add("上海");
		allAreas.add("天津");
		allAreas.add("广州");
		allAreas.add("深圳");
		allAreas.add("成都");
		allAreas.add("杭州");
		allAreas.add("大连");

		// 选择的电台集合
		List<String> selects = new ArrayList<>();
		// 存放遍历过程中某电台覆盖地区集合 和 所有没有覆盖地区集合 的交集
		Set<String> tempSet = new HashSet<>();
		// maxKey 存放内次遍历包含最多的未覆盖地区的电台的 key
		String maxKey;

		while (allAreas.size() > 0) {
			maxKey = null;

			for (String key : broadcasts.keySet()) {
				tempSet.clear();

				HashSet<String> areas = broadcasts.get(key);
				tempSet.addAll(areas);
				// tempSet 和 allAreas 取差集 赋到 tempSet（key 电台地区对应的所有区域差集）
				tempSet.retainAll(allAreas);

				Set<String> maxSet = new HashSet<>();
				if (maxKey != null) {
					maxSet = broadcasts.get(maxKey);
					maxSet.retainAll(allAreas);
				}

				// tempSet.size() > maxSet.size() 体现出贪心算法的特点,每次都选择最优的
				if (tempSet.size() > 0 && (maxKey == null || tempSet.size() > maxSet.size())) {
					maxKey = key;
				}
			}
			if (maxKey != null) {
				selects.add(maxKey);
				// 将 maxKey 指向的广播电台覆盖的地区，从 allAreas 去掉
				allAreas.removeAll(broadcasts.get(maxKey));
			}
		}
		System.out.println("得到的选择结果是" + selects);//[K1,K2,K3,K5]
	}
}
